#!/usr/bin/env bash
Loss=wing
MODELDIR="./models/checkpoint/model-$Loss"
mkdir -p "$MODELDIR"
LOGFILE="$MODELDIR/log"

CUDA_VISIBLE_DEVICES='0' python -u train.py\
  --dataroot "/data/datasets/face-ldmk/PFPLD-Dataset/train_data/list.txt" \
  --val_dataroot "/data/datasets/face-ldmk/PFPLD-Dataset/test_data/list.txt" \
  --step "100,200" \
  --end_epoch 400 \
  --snapshot "$MODELDIR" \
  --tensorboard "$MODELDIR/tensorboard" \
  --resume  "" \
  --backbone PFLDResNet \
  --loss "$Loss" \
  --workers 8 \
  --base_lr 1e-2 \
  --train_batchsize 256 \
  --val_batchsize 8 \
  > "$LOGFILE" 2>&1 &
